//
// * Copyright (c) 2002-2009 "Neo Technology,"
// *     Network Engine for Objects in Lund AB [http://neotechnology.com]
// *
// * This file is part of Neo4j.
// * 
// * Neo4j is free software: you can redistribute it and/or modify
// * it under the terms of the GNU Affero General Public License as
// * published by the Free Software Foundation, either version 3 of the
// * License, or (at your option) any later version.
// * 
// * This program is distributed in the hope that it will be useful,
// * but WITHOUT ANY WARRANTY; without even the implied warranty of
// * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// * GNU Affero General Public License for more details.
// * 
// * You should have received a copy of the GNU Affero General Public License
// * along with this program. If not, see <http://www.gnu.org/licenses/>.
// 
namespace org.neo4j.kernel.impl
{

	using Test = junit.framework.Test;
	using TestCase = junit.framework.TestCase;
	using TestSuite = junit.framework.TestSuite;

	using GraphDatabaseService = org.neo4j.graphdb.GraphDatabaseService;
	using Transaction = org.neo4j.graphdb.Transaction;
	using EmbeddedGraphDatabase = org.neo4j.kernel.EmbeddedGraphDatabase;
	using NodeManager = org.neo4j.kernel.impl.core.NodeManager;
	using TestNeo4j = org.neo4j.kernel.impl.core.TestNeo4j;

	public abstract class AbstractNeo4jTestCase : TestCase
	{
		protected internal const string NEO4J_BASE_PATH = "target/var/";

		public AbstractNeo4jTestCase(string testName) : base(testName)
		{
		}

		private GraphDatabaseService graphDb;
		private Transaction tx;

		public virtual GraphDatabaseService getGraphDb()
		{
			return graphDb;
		}

		public virtual EmbeddedGraphDatabase getEmbeddedGraphDb()
		{
			return (EmbeddedGraphDatabase) graphDb;
		}

		public virtual Transaction getTransaction()
		{
			return tx;
		}

		public static Test suite()
		{
			TestSuite suite = new TestSuite(typeof(TestNeo4j));
			return suite;
		}

		public static string getStorePath(string endPath)
		{
			return NEO4J_BASE_PATH + endPath;
		}

		public virtual void setUp()
		{
			graphDb = new EmbeddedGraphDatabase(getStorePath("neo-test"));
			tx = graphDb.beginTx();
		}

		public virtual void tearDown()
		{
			if (tx != null)
			{
				tx.finish();
			}
			graphDb.shutdown();
		}

		public virtual void setTransaction(Transaction tx)
		{
			this.tx = tx;
		}

		public virtual void newTransaction()
		{
			if (tx != null)
			{
				tx.success();
				tx.finish();
			}
			tx = graphDb.beginTx();
		}

		public virtual void commit()
		{
			if (tx != null)
			{
				tx.success();
				tx.finish();
				tx = null;
			}
		}

		public virtual NodeManager getNodeManager()
		{
			return ((EmbeddedGraphDatabase) graphDb).getConfig().getNeoModule().getNodeManager();
		}
	}
}